
import os

from huggingface_hub import snapshot_download

from config import LLM_CACH_DIR

if __name__=="__main__":
    model_dir = snapshot_download(repo_id="THUDM/glm-4-9b-chat",
                                  local_dir=os.path.join(LLM_CACH_DIR,"THUDM/glm-4-9b-chat"),
                                  local_dir_use_symlinks=False)
    print(model_dir)